import torch

if torch.cuda.is_available():
    print(f"CUDA版本：{torch.version.cuda}")
    print(f"CUDA设备数量：{torch.cuda.device_count()}")
    print(f"当前设备：{torch.cuda.get_device_name(0)}")
else:
    print("未检测到CUDA设备，可能未安装或未启用GPU支持")